#ifndef INVOICESTATISTICS_H
#define INVOICESTATISTICS_H

// 将 QMainWindow 改为 QWidget
#include <QWidget>
#include <QSqlDatabase>
#include <QSqlQuery>
#include <QSqlError>
#include <QDateEdit>
#include <QPushButton>
#include <QTableView>
#include <QLabel>
#include <QSqlTableModel>
#include <QHBoxLayout>
#include <QVBoxLayout>
#include <QGroupBox>
#include <QMessageBox>

// 继承自 QWidget 而不是 QMainWindow
class InvoiceStatistics : public QWidget
{
    Q_OBJECT

public:
    InvoiceStatistics(QWidget *parent = nullptr);
    ~InvoiceStatistics();

private slots:
    void searchInvoices();

private:
    bool setupDatabase();

    // UI 元素
    QDateEdit *startDateEdit;
    QDateEdit *endDateEdit;
    QPushButton *searchButton;
    QTableView *invoiceTableView;
    QLabel *totalInvoicesLabel;
    QLabel *totalAmountLabel;

    // 数据库
    QSqlDatabase db;
    QSqlTableModel *model;
};

#endif // INVOICESTATISTICS_H
